package com.itany.corejava.code13_jdbc;

import javax.xml.crypto.Data;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * @author Miss Chen
 * @version 1.0
 * @date 2024年08月07日14:57
 */
public class Test06_保存返回主键 {
    public static void main(String[] args) throws DataAccessException {
        User user=new User("zhao2","2222","ssss","asasasas");
        Integer id=insertReturnPrimaryKey(user);
        System.out.println(id);

    }

    public static Integer insertReturnPrimaryKey(User user) throws DataAccessException {
        Connection conn=null;
        PreparedStatement ps=null;
        Integer id=null;
        ResultSet rs=null;
        try {
            conn=JDBCUtil.getConnection();
            String sql=new StringBuffer()
                    .append(" insert into ")
                    .append(" t_user ")
                    .append(" (username,password,phone,address) ")
                    .append(" values ")
                    .append(" (?,?,?,?) ")
                    .toString();
            // 需要返回主键值
            // 在获取状态集的时候，需要设置状态集操作时保存返回主键
            // 第二个参数，用于设置保存是否返回自动生成的主键的标志
            // 其值是PreparedStatement继承自Statement提供的两个常量值
            // RETURN_GENERATED_KEYS：返回自动生成的键
            // No_GENERATED_KEYS：不返回自动生成的键

            ps=conn.prepareStatement(sql,PreparedStatement.RETURN_GENERATED_KEYS);
            ps.setString(1,user.getUsername());
            ps.setString(2,user.getPassword());
            ps.setString(3,user.getPhone());
            ps.setString(4,user.getAddress());
            //执行DML操作
            ps.executeUpdate();

            //当执行DML操作之后，数据库中会生成对应的主键的值
            // 此时需要获取到生成后的主键的值
            // Statement提供了获取自动生成的键的方法
            // 返回值是一个主键值，但是我们对其返回的列名可能并不知道
            rs=ps.getGeneratedKeys();

            while(rs.next()){

                //由于生成的主键是由数据库自动操作的
                // 对于开发者而言，数据库厂商为其定义的列名我们可能并不知道
                //如何映射?
                // 虽然开发者不知道列名，但是能够确定的是返回的值有且仅有一列
                //因此可以通过索引号进行映射，直接将第一列数据映射给对应的主键值
                id=rs.getInt(1);
            }

            return id;

        } catch (DataAccessException e) {
            e.printStackTrace();
            throw e;
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("数据库访问异常");
        } finally {
            JDBCUtil.close(conn,ps,rs);
        }


    }
}
